# -*- coding:utf-8 -*-
__author__ = 'root'
import os
from kaoala_search.settings import BASE_DIR
import datetime
import shutil


base_path=os.path.join(BASE_DIR,'kaoala_search_data')  #基础路径
kaoala_jieba_dict=os.path.join(base_path,'kaoala_dict.txt')    #kaoala分词词汇文件路径

#---------------源数据路径-----------------------
source_base_path=os.path.join(base_path,'source_data')   #源数据基础路径
current_source_path=os.path.join(source_base_path,'current_data')   #当前最新的源数据
#-------------end--------------------------------------


#----------------前台数据的路径-----------------------------------
reception_base_path=os.path.join(base_path,'reception_data')   #前台基础路
reception_user_search_data=os.path.join(reception_base_path,'user_search_data')    #用于用户搜索词汇的记录
#--------------------end------------------------------



def source_path_handle():
    '''
    源数据文件夹的处理，备份源数据之前的结果
    :return:
    '''
    if os.path.exists(base_path)==False:   #判断目录是否存在
        os.makedirs(base_path)
    if os.path.exists(current_source_path)==True:
        one_path,two_path=os.path.split(current_source_path)
        two_path=two_path+datetime.datetime.now().strftime('%Y%m%d%H%M%S')
        new_path=os.path.join(one_path,two_path)
        shutil.copytree(current_source_path,new_path)   #目录备份
        shutil.rmtree(current_source_path)   #删除文件夹
        os.makedirs(current_source_path)   #新建文件夹
    else:
        os.makedirs(current_source_path)


def reception_path_handle():
    '''
    前台数据的文件夹处理，备份用户搜索词汇的记录结果
    :return:
    '''
    if os.path.exists(reception_base_path)==False:   #判断目录是否存在
        os.makedirs(reception_base_path)
    if os.path.exists(reception_user_search_data)==True:
        one_path,two_path=os.path.split(reception_user_search_data)
        two_path=two_path+datetime.datetime.now().strftime('%Y%m%d%H%M%S')
        new_path=os.path.join(one_path,two_path)
        shutil.copytree(reception_user_search_data,new_path)   #目录备份
    else:
        os.makedirs(reception_user_search_data)


def source_file_copy():
    '''
    把源数据处理好的文件复制到前台的数据文件夹中
    :return:
    '''
    file_list = os.listdir(current_source_path)  #列出目录下的所有文件和目录
    for sub_file in file_list:
        if os.path.isfile(os.path.join(current_source_path,sub_file))==True:
            shutil.copyfile(os.path.join(current_source_path,sub_file),os.path.join(reception_base_path,sub_file))



